
# 定义登录功能

user_data = [{'role':'admin','account':'admin','password':'123456','dept':'tester'},{'role':'test','account':'test','password':'123456','dept':'tester'}]

# 定义默认返回结果
result = {'code':0 , 'message':''}


# 登录功能
def login(username,password):

    # 如果用户名为空或密码为空 ，给出用户名或密码不能为空的提示
    if username is None or username == '':
        result.update({'code':1,'message':"用户名不能为空!"})
        return result
    if password is None or password == '':
        result.update({'code':2,'message':'密码不能为空!'})
        return result

    # 用户和密码匹配成功，登录成功 ，并且将用户的数据一起返回 。
    for user_info in user_data:
        if username == user_info.get('account') and password == user_info.get('password'):
            result.update({"code":0,"message":"登录成功!",'user_list':user_data})
            return result

    # 如果用户和密码不匹配 ，给出错误提示 。
    result.update({'code':3,"message":"用户名或密码不正确，请重新输入!"})
    return result


if __name__ == '__main__':
    username = input("请输入用户名:")
    password = input("请输入密码:")

    login_result = login(username,password)
    print(login_result)